package com.trace.demo.repository;

import com.trace.demo.entity.User;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.Pageable;
import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.data.jpa.repository.Query;
import org.springframework.stereotype.Repository;

import java.time.LocalDateTime;
import java.util.List;
import java.util.Optional;

/**
 * 用户数据访问接口
 */
@Repository
public interface UserRepository extends JpaRepository<User, Long> {

    /**
     * 根据用户名查找用户
     * @param username 用户名
     * @return 用户对象
     */
    Optional<User> findByUsername(String username);

    /**
     * 根据用户名和密码查找用户
     * @param username 用户名
     * @param password 密码
     * @return 用户对象
     */
    Optional<User> findByUsernameAndPassword(String username, String password);

    /**
     * 检查用户名是否存在
     * @param username 用户名
     * @return 是否存在
     */
    boolean existsByUsername(String username);

    /**
     * 根据角色查找用户
     * @param role 角色
     * @return 用户列表
     */
    List<User> findByRole(User.UserRole role);
    
    /**
     * 根据角色查找用户（分页）
     * @param role 角色
     * @param pageable 分页参数
     * @return 分页用户列表
     */
    Page<User> findByRole(User.UserRole role, Pageable pageable);
    
    /**
     * 根据角色统计用户数量
     * @param role 角色
     * @return 该角色的用户数量
     */
    long countByRole(User.UserRole role);

    /**
     * 查找指定时间之后创建的用户
     * @param createdAt 创建时间
     * @return 用户列表
     */
    List<User> findByCreatedAtAfter(LocalDateTime createdAt);

    /**
     * 统计每个角色的用户数量
     * @return 角色和用户数量的列表
     */
    @Query("SELECT u.role as role, COUNT(u) as count FROM User u GROUP BY u.role")
    List<Object[]> countUsersByRole();

    /**
     * 统计今天新增的用户数量
     * @return 新增用户数量
     */
    @Query("SELECT COUNT(u) FROM User u WHERE DATE(u.createdAt) = CURRENT_DATE")
    long countTodayNewUsers();

    /**
     * 根据用户名模糊查询
     * @param username 用户名关键词
     * @return 用户列表
     */
    List<User> findByUsernameContaining(String username);
    
    /**
     * 根据用户名模糊查询（分页）
     * @param username 用户名关键词
     * @param pageable 分页参数
     * @return 分页用户列表
     */
    Page<User> findByUsernameContaining(String username, Pageable pageable);

    /**
     * 根据用户名模糊查询
     * @param username 用户名关键词
     * @return 用户列表
     */
    @Query("SELECT u FROM User u WHERE u.username LIKE %:username%")
    List<User> searchByUsername(String username);
    
    /**
     * 根据状态查找用户
     * @param status 状态值 (1:启用, 0:禁用)
     * @return 用户列表
     */
    List<User> findByStatus(Integer status);
    
    /**
     * 根据状态查找用户（分页）
     * @param status 状态值 (1:启用, 0:禁用)
     * @param pageable 分页参数
     * @return 分页用户列表
     */
    Page<User> findByStatus(Integer status, Pageable pageable);
    
    /**
     * 根据角色和状态查找用户
     * @param role 角色
     * @param status 状态值
     * @return 用户列表
     */
    List<User> findByRoleAndStatus(User.UserRole role, Integer status);
    
    /**
     * 根据角色和状态查找用户（分页）
     * @param status 状态值
     * @param role 角色
     * @param pageable 分页参数
     * @return 分页用户列表
     */
    Page<User> findByStatusAndRole(Integer status, User.UserRole role, Pageable pageable);
    
    /**
     * 统计启用状态的用户数量
     * @return 启用状态的用户数量
     */
    long countByStatus(Integer status);

    Optional<User> findByEmail(String email);
    
    Optional<User> findByPhone(String phone);
    
    List<User> findByUserType(String userType);
    
    Long countByUserType(String userType);
    
    @Query("SELECT COUNT(u) FROM User u")
    Long countUsers();
    
    @Query("SELECT COUNT(u) FROM User u WHERE u.userType = 'FARMER'")
    Long countFarmers();
    
    @Query("SELECT COUNT(u) FROM User u WHERE u.userType = 'DEALER'")
    Long countDealers();
}
